import router from '@/router'
import { useUserStore, usePermissionStore } from '@/store'
import NProgress from '@/utils/nprogress'
import { RouteRecordRaw } from 'vue-router'
import { TOKEN_KEY } from '@/enums/CacheEnum'

export function setupPermission() {
    // 白名单路由
    const whiteList = ['/login']
    /**
     * 1 判断token是否存在
     * 2 不存在token 则判断是否在白名单中
     *    2.1 在白名单中 则跳转
     *    2.2 不在白名单中,则跳转登录(退出存在redirect、未登录时跳转)
     * 3 存在token
     * 3.1 如何跳转至登录页面,则跳转首页
     * 3.2 判断是否有权限权限roles
     * 3.3 有权限roles
     *     3.3.1 未匹配到任何路由，跳转404或from
     *     3.3.2 跳转next();
     * 3.4 无权限roles(刷新时)
     *     3.4.1 获取用户的权限roles
     *     3.4.2 获取对应的动态路由权限accessRoutes，并添加
     *     3.4.3 重新跳转至该路由 next({ ...to, replace: true });
     */

    /**
     * 注意点
     * 当路由跳转不是next()或 next({ ...to, replace: true })时，需要NProgress.done()
     */
    router.beforeEach(async (to, from, next) => {
        NProgress.start()
        const hasToken = localStorage.getItem(TOKEN_KEY)
        if (hasToken) {
            if (to.path === '/login') {
                // 如果已登录，跳转首页
                next({ path: '/' })
                NProgress.done()
            } else {
                const userStore = useUserStore()
                const hasRoles =
                    userStore.user.roles && userStore.user.roles.length > 0
                if (hasRoles) {
                    // 未匹配到任何路由，跳转404
                    if (to.matched.length === 0) {
                        from.name ? next({ name: from.name }) : next('/404')
                    } else {
                        next()
                    }
                } else {
                    const permissionStore = usePermissionStore()
                    try {
                        const { roles } = await userStore.getUserInfo()
                        const accessRoutes =
                            await permissionStore.generateRoutes(roles)
                        accessRoutes.forEach((route: RouteRecordRaw) => {
                            router.addRoute(route)
                        })
                        next({ ...to, replace: true })
                    } catch (error) {
                        // 移除 token 并跳转登录页
                        await userStore.resetToken()
                        next(`/login?redirect=${to.path}`)
                        NProgress.done()
                    }
                }
            }
        } else {
            // 未登录可以访问白名单页面
            if (whiteList.indexOf(to.path) !== -1) {
                next()
            } else {
                next(`/login?redirect=${to.path}`)
                NProgress.done()
            }
        }
    })

    router.afterEach(() => {
        NProgress.done()
    })
}
